#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
配置设置脚本
帮助用户快速从模板创建配置文件
"""

import os
import shutil
import sys


def setup_config():
    """设置配置文件"""
    
    template_file = "config_template.py"
    config_file = "config.py"
    
    print("🔒 RocketMQ 配置设置")
    print("=" * 40)
    
    # 检查模板文件是否存在
    if not os.path.exists(template_file):
        print(f"❌ 错误: 找不到模板文件 {template_file}")
        return False
    
    # 检查配置文件是否已存在
    if os.path.exists(config_file):
        print(f"⚠️  配置文件 {config_file} 已存在")
        response = input("是否要覆盖? (y/N): ").strip().lower()
        if response != 'y':
            print("❌ 操作已取消")
            return False
    
    try:
        # 复制模板文件
        shutil.copy2(template_file, config_file)
        print(f"✅ 成功创建配置文件: {config_file}")
        
        # 显示下一步操作说明
        print("\n📝 下一步操作:")
        print(f"1. 编辑 {config_file} 文件")
        print("2. 填入您的配置信息:")
        print("   - namesrv_addr: RocketMQ NameServer 地址")
        print("   - access_key: ACL 访问密钥（可选）")
        print("   - secret_key: ACL 密钥（可选）")
        print("   - 其他配置项...")
        print("\n3. 运行测试: python test_connection.py")
        
        return True
        
    except Exception as e:
        print(f"❌ 创建配置文件失败: {e}")
        return False


def main():
    """主函数"""
    
    print("🚀 RocketMQ Python 客户端配置设置")
    print("=" * 40)
    
    # 设置配置文件
    config_ok = setup_config()
    
    print("\n" + "=" * 40)
    
    if config_ok:
        print("✅ 配置设置完成！")
        print("\n💡 安全提示:")
        print("- 配置文件不会被提交到版本控制系统")
        print("- 请妥善保管您的敏感信息")
    else:
        print("⚠️  配置设置失败，请检查上述信息")
        return 1
    
    return 0


if __name__ == "__main__":
    sys.exit(main()) 